package com.zsr.servlet;

import org.apache.commons.io.IOUtils;
import sun.misc.BASE64Encoder;

import javax.servlet.ServletContext;
import javax.servlet.ServletException;
import javax.servlet.http.HttpServlet;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.net.URLEncoder;

public class Download extends HttpServlet {
    @Override
    protected void doGet(HttpServletRequest req, HttpServletResponse resp) throws ServletException, IOException {
        //1. 获取要下载的文件名
        String downloadFileName = "test.png";

        //2.读取要下载的文件内容（通过ServletContext对象读取）
        ServletContext servletContext = getServletContext();

        //获取下载到的文件类型
        String mimeType = servletContext.getMimeType("/file/" + downloadFileName);
        System.out.println(mimeType);

        //4.在回传前，通过响应头告诉客户端返回的数据类型
        resp.setContentType(mimeType);

        //5. 告诉服务器收到的数据是用于下载（还是使用响应头，不然只显示图片而不是下载）
        //Content-Disposition 响应头指示回复的内容该以何种形式展示，是以内联的形式（即网页或者页面的一部分），还是以附件的形式下载并保存到本地。
        //attachment 附件
        //filename 可以和工程内的文件名不一样，是客户端下载后的文件名
//        resp.setHeader("Content-Disposition", "attachment; filename=" + downloadFileName);

        String ua = req.getHeader("User-Agent");
        // 判断是否是火狐浏览器
        if (ua.contains("Firefox")) {
            //火狐浏览器使用Base64编码
            resp.setHeader("Content-Disposition", "attachment; fileName=" + "=?utf-8?B?" + new BASE64Encoder().encode((
                    "中国.png").getBytes("utf-8")) + "?=");
        } else {
            //谷歌IE浏览器 可以设置编码为UTF-8  url编码是把汉字转换为%xx%xx的格式
            resp.setHeader("Content-Disposition", "attachment; filename=" + URLEncoder.encode("中国.png", "UTF-8"));
        }


        InputStream resourceAsStream = servletContext.getResourceAsStream("/file/" + downloadFileName);
        //获取响应的输出流
        OutputStream outputStream = resp.getOutputStream();
        //3.把下载的文件内容回传给客户端
        //读取输入流全部的数据，复制给输出流，输出给客户端
        IOUtils.copy(resourceAsStream, outputStream);

    }
}
